package com.cheng.dataStructure.stack;

import java.util.ArrayList;
import java.util.List;
import java.util.Stack;

/**
 * @User Administrator
 * @Classname PolandNotation
 * @Project Data-structure-and-algorithm
 * @Description 代码实现逆波兰计算器
 *思路：
 * 1. 将 "3 4 + 5 * 6 -" 放入ArrayList中
 * 2. 将 ArrayList 传递给一个方法，遍历 ArrayList ，配合栈，完成计算
 * @Author wpcheng
 * @Create 2021-07-21-10:42
 * @Version 1.0
 */
public class PolandNotation {
    public static void main(String[] args) {
        //定义逆波兰表达式
        String suffixExpression = "30 4 + 5 * 6 -";
        List<String> listString = getListString(suffixExpression);
        System.out.println(listString);
        int calculate = calculate(listString);
        System.out.println("计算结果为：");
        System.out.println(calculate);

    }

    /**
     * 思路：
     * 1. 将 "3 4 + 5 * 6 -" 放入ArrayList中
     * @param suffixExpression 逆波兰表达式
     * @return
     */
    public static List<String> getListString(String suffixExpression){
        //将 suffixExpression 按照空格分割,返回一个字符串组
        String[] split = suffixExpression.split(" ");
        ArrayList<String> list = new ArrayList<>();
        //把将逆波兰表达式的数和符号依次放入 ArrayList中
        for (String ele : split) {
            list.add(ele);
        }
        return list;
    }
    //2. 将 ArrayList 传递给一个方法，遍历 ArrayList ，配合栈，完成计算
    public static int calculate(List<String> ls){
        Stack<String> stack = new Stack<>();
        //遍历ls
        for (String item : ls) {
            //如果遍历出的是多位数
            if (item.matches("\\d+")){
                //就入栈
                stack.push(item);
                //如果遍历出的是运算符
            }else {
                //从栈中pop出两个数，进行运算，然后再入栈
                int num2 = Integer.parseInt(stack.pop());
                int num1 = Integer.parseInt(stack.pop());
                int res = 0;
                if (item.equals("+")){
                    res = num1 + num2;
                }else if (item.equals("-")){
                    res = num1 - num2;
                }else if (item.equals("*")){
                    res = num1 * num2;
                }else if (item.equals("/")){
                    res = num1 / num2;
                }else {
                    throw new RuntimeException("运算符有误");
                }
                //把结果入栈
                stack.push(""+res);
            }
        }
        //把结果取出来
        return Integer.parseInt(stack.pop());
    }

}
